from easydict import EasyDict
import torch

cfg = EasyDict()

cfg.check_path = "checkpoint"
cfg.num_class = 37 
cfg.resize = 224
cfg.batch_size = 48
cfg.lr = 0.001
cfg.warmup = 10
cfg.min_lr = 0.0005
cfg.cyc = 20
cfg.resume = 0
cfg.epochs = 300
cfg.freeze = False
cfg.ct_weight = 0.01
cfg.neg_weight = 10.0
cfg.device = torch.device('cuda' if torch.cuda.is_available() else "cpu")
cfg.gpu_idx = [1, 2]

cfg.dataset_path = "/data/public/The_Oxford-IIIT_Pet_Dataset/imagesByClass/"


cfg.test_dataPath = './' + cfg.check_path + '/testData'
cfg.csv_file = cfg.check_path + "/dataset.csv"
cfg.pth = "/data/user/yangyaotian/Test/swinT/pth/swin_base_patch4_window12_384_22k.pth"
cfg.best_model_path = cfg.check_path + "/best_model_checkpoint.pth.tar"
cfg.result_loss = cfg.check_path + "/result_loss.png"
cfg.result_acc = cfg.check_path + "/result_acc.png"
cfg.result_lr = cfg.check_path + "/result_lr.png"
